package com.mjf.spark.day03

import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}

/**
 * 转换算子-glom
 *    将RDD一个分区中的元素，组合成一个新的数组
 */
object Spark04_Transformation_glom {
  def main(args: Array[String]): Unit = {

    // 创建SparkConf配置文件
    val conf = new SparkConf().setMaster("local[*]").setAppName("Spark04_Transformation_glom")
    // 创建SparkContext对象
    val sc = new SparkContext(conf)

    val rdd: RDD[Int] = sc.makeRDD(List(1, 2, 3, 4, 5, 6), 2)

    println("------没有glom之前------")
    rdd.mapPartitionsWithIndex((index, datas) => {
      println(index + "--->" + datas.mkString(","))
      datas
    }).collect()

    println("------没有glom之前------")
    val newRDD: RDD[Array[Int]] = rdd.glom()

    newRDD.mapPartitionsWithIndex((index, datas) => {
      println(index + "--->" + datas.next().mkString(","))
      datas
    }).collect()

    // 关闭连接
    sc. stop()

  }
}
